load("//bazel:api.bzl", "mojo_test")

package(default_visibility = ["//max:consumers"])

# All SM100 tests require B200 GPU
[
    mojo_test(
        name = src + ".test",
        size = "large",
        srcs = [src],
        enable_assertions = True,
        exec_properties = {
            "test.resources:gpu-memory": "2",
        },
        tags = ["gpu"],
        target_compatible_with = [
            "//:has_gpu",
            "//:b200_gpu",
        ],
        deps = [
            "//max:internal_utils",
            "//max:linalg",
            "@mojo//:stdlib",
            "@mojo//:test_utils",
        ],
    )
    for src in glob(["*.mojo"])
]
